package club.xiaojiawei.greedy;

/**
 * @author 肖嘉威
 * @version 1.0
 * @date 5/30/22 10:36 PM
 * @question 714. 买卖股票的最佳时机含手续费
 * @description 给定一个整数数组 prices，其中 prices[i]表示第 i 天的股票价格 ；整数 fee 代表了交易股票的手续费用。
 * 你可以无限次地完成交易，但是你每笔交易都需要付手续费。如果你已经购买了一个股票，在卖出它之前你就不能再继续购买股票了。
 * 返回获得利润的最大值。
 * 注意：这里的一笔交易指买入持有并卖出股票的整个过程，每笔交易你只需要为支付一次手续费。
 */
public class MaxProfit714 {

    public static void main(String[] args) {
        MaxProfit714 test = new MaxProfit714();
        int result = test.maxProfit3(new int[]{1, 4, 4, 8}, 2);
        System.out.println(result);
    }

    /**
     * 贪心
     * 时间复杂度O(n)
     * 空间复杂度O(1)
     * @param prices
     * @param fee
     * @return
     */
    public int maxProfit(int[] prices, int fee) {
        int min = prices[0], max = prices[0];
        int result = 0;
        for (int i = 1; i < prices.length; i++) {
            if (prices[i] > max){
                max = prices[i];
                if (i == prices.length - 1 && max - min - fee > 0){
                    result += max - min - fee;
                    max = min = 0;
                }
            }else if (max > prices[i] + fee){
                if (max - min - fee > 0){
                    result += max - min - fee;
                    max = min = prices[i];
                }else if (prices[i] < min){
                    max = min = prices[i];
                }
            }else if (prices[i] < min){
                min = max = prices[i];
            }
        }
        if (max - min - fee > 0){
            result += max - min - fee;
        }
        return result;
    }

    /**
     * 官方-dp
     * 时间复杂度O(n)
     * 空间复杂度O(n)或O(1)
     * @param prices
     * @param fee
     * @return
     */
    public int maxProfit2(int[] prices, int fee) {
        int n = prices.length;
//        sell：抛售股票后剩下的钱；buy：买入一只股票后剩下的钱
        int sell = 0, buy = -prices[0];
        for (int i = 1; i < n; ++i) {
            sell = Math.max(sell, buy + prices[i] - fee);
            buy = Math.max(buy, sell - prices[i]);
        }
        return sell;
    }

    /**
     * 官方-贪心（nb）
     * 时间复杂度O(n)
     * 空间复杂度O(1)
     * @param prices
     * @param fee
     * @return
     */
    public int maxProfit3(int[] prices, int fee) {
        int n = prices.length;
        int buy = prices[0] + fee;
        int profit = 0;
        for (int i = 1; i < n; ++i) {
            if (prices[i] + fee < buy) {
                buy = prices[i] + fee;
            } else if (prices[i] > buy) {
                profit += prices[i] - buy;
                buy = prices[i];
            }
        }
        return profit;
    }
}
